absl-py>=0.12.0
clu>=0.0.4
flax>=0.4.1
jax>=0.4.3
ml-collections>=0.1.1
numpy>=1.21.5
optax>=0.0.6
scikit-learn>=1.0.2
scipy>=1.6.3
sentencepiece>=0.1.95
tensorflow>=2.8.0
tensorflow-datasets>=4.5.2
